import functools


def set_func(func):
    @functools.wraps(func)
    def wrapper():
        return func()

    return wrapper


@set_func
def test1():
    print('test1')


@set_func
def test2():
    print('test2')


if __name__ == '__main__':
    print(test1.__name__)
    print(test2.__name__)
